package com.lwl.leetcode.滑动窗口;

import com.lwl.Utils.PrintUtil;

import java.util.HashMap;
import java.util.LinkedList;
import java.util.Map;

/**
 * @author lwl
 * @Description 496. 下一个更大元素 I
 * 简单
 * 相关标签
 * conpanies icon
 * 相关企业
 * nums1 中数字 x 的 下一个更大元素 是指 x 在 nums2 中对应位置 右侧 的 第一个 比 x 大的元素。
 * 给你两个 没有重复元素 的数组 nums1 和 nums2 ，下标从 0 开始计数，其中nums1 是 nums2 的子集。
 * 对于每个 0 <= i < nums1.length ，找出满足 nums1[i] == nums2[j] 的下标 j ，并且在 nums2 确定 nums2[j] 的 下一个更大元素 。如果不存在下一个更大元素，那么本次查询的答案是 -1 。
 * 返回一个长度为 nums1.length 的数组 ans 作为答案，满足 ans[i] 是如上所述的 下一个更大元素
 * @date 2025/9/1 15:50
 */
public class Code496NextGreaterElement {
    public static void main(String[] args) {
        int[] nums1 = {2,4};
        int[] nums2 = {1,2,3,4};
        int[] res = nextGreaterElement(nums1, nums2);
        PrintUtil.printArray(res);
    }

    public static int[] nextGreaterElement(int[] nums1, int[] nums2) {
        // 保存 nums 中每个位置左边第一个比他大的下标
        Map<Integer, Integer> map = new HashMap<>();
        // 从左往右递减
        LinkedList<Integer> queue = new LinkedList<>();
        for (int i = 0; i < nums2.length; i++) {
            while (!queue.isEmpty() && nums2[i] > nums2[queue.peekLast()]) {
                Integer curLast = queue.pollLast();
                map.put(nums2[curLast], i);
            }
            queue.addLast(i);
        }
        int[] res = new int[nums1.length];
        for (int i = 0; i < nums1.length; i++) {
            Integer index = map.get(nums1[i]);
            if (index == null) {
                res[i] = -1;
            } else {
                res[i] = nums2[index];
            }
        }
        return res;
    }
}
